from page.base import Base
from selenium.webdriver.common.by import By

from page.login import Login
from page.register import Register

class Main(Base):
    _base_url = "https://work.weixin.qq.com/"
    def goto_register(self):
        self.find(By.CSS_SELECTOR,'.index_head_info_pCDownloadBtn').click()
        return Register(self._driver)
    def goto_login(self):
        self.find(By.XPATH, '//*[@id="indexTop"]//*[@class="index_top_operation_loginBtn"]').click()
        return Login(self._driver)
